
from train import Model
import torch

# 实例化模型
model = Model()
# 加载模型
model.load_state_dict(torch.load("./save/a.pth"))
# 模型调整为验证模式
model.eval()
